package com.own.system.utils;

import com.own.system.entity.Permission;
import com.own.system.vo.RouterVo;
import org.springframework.beans.BeanUtils;

import java.util.ArrayList;
import java.util.List;
import java.util.Objects;
import java.util.Optional;

/**
 * 生成菜单树
 */
public class MenuTree {
    /**
     * 生成路由
     *
     * @param menuList 菜单列表
     * @param pid      父级菜单
     * @return 返回静态路由列表
     */
    public static List<RouterVo> makeRouter(List<Permission> menuList, long pid) {
        //创建集合保存路由列表
        List<RouterVo> routerVoList = new ArrayList<RouterVo>();
        //如果menuList不为空,则使用菜单列表，否则创建集合对象
        Optional.ofNullable(menuList).orElse(new ArrayList<Permission>())
                //筛选不为空的菜单以及菜单父id相同的数据
                .stream().filter(item -> item != null && item.getParentId() == pid).forEach(item -> {
                    //创建路由对象
                    RouterVo routerVo = new RouterVo();
                    routerVo.setName(item.getName());//设置路由名称
                    routerVo.setPath(item.getPath());//设置路由路径
                    //判断是否是一级菜单
                    if (item.getParentId() == 0L) {
                        routerVo.setComponent("Layout");//一级菜单组件
                        routerVo.setAlwaysShow(true);//显示路由
                    } else {
                        routerVo.setComponent(item.getUrl());//具体组件
                        routerVo.setAlwaysShow(false);//折叠路由
                    }
                    //设置meta信息
                    routerVo.setMeta(routerVo.new Meta(item.getLabel(), item.getIcon(), item.getCode().split(",")));
                    //递归调用生成子路由
                    List<RouterVo> children = makeRouter(menuList, item.getId());
                    routerVo.setChildren(children);//设置子路由到路由对象
                    //将路由添加到路由列表中
                    routerVoList.add(routerVo);
                });
        return routerVoList;
    }

    public static List<Permission> makeMenuTree(List<Permission> menuList, long pid) {
        //创建集合保存路由列表
        List<Permission> permissionList = new ArrayList<Permission>();
        //如果menuList不为空,则使用菜单列表，否则创建集合对象
        Optional.ofNullable(menuList).orElse(new ArrayList<Permission>()).stream().filter(item -> item != null && Objects.equals(item.getParentId(), pid))
                .forEach(item -> {
                    //创建菜单权限对象
                    Permission permission = new Permission();
                    //复制属性
                    BeanUtils.copyProperties(item, permission);
                    //获取每一个item的下级菜单，递归生成菜单树
                    List<Permission> children = makeMenuTree(menuList, item.getId());
                    //设置子菜单
                    permission.setChildren(children);
                    //将菜单添加到菜单列表中
                    permissionList.add(permission);
                });
        return permissionList;
    }
}
